package com.trust.pams.util;

import java.io.File;
import java.util.List;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.read.builder.ExcelReaderBuilder;
import com.alibaba.excel.read.builder.ExcelReaderSheetBuilder;

/**
 * 读取量超过1000行的excel，建议用这个工具类，效率比poi高，内存占用也很小。
 * 具体解析可参照   github    https://github.com/alibaba/easyexcel
 * @author hchen
 *
 */
public class EasyExcelUtil {

	
	public static List<Object> analysisExcel(String excelFilePath){
		if(DataUtil.isEmpty(excelFilePath)) {
		}
		if(excelFilePath.toLowerCase().endsWith(".xls") && excelFilePath.toLowerCase().endsWith(".xlsx")) {
		}
		
		ExcelListener<Object> listener=new ExcelListener<Object>();
		File excelFile=new File(excelFilePath);
			
		EasyExcel.read(excelFile, listener).sheet(0).doRead();
		
		List<Object> objList=listener.getDatas();
		return objList;
	}
	
	
	/**
	 * 读取单个sheet或只有一个sheet的excel文件。
	 * @param excelFilePath excel文件
	 * @param listener excel监听器
	 * @param sheetNum  按sheetNum加载 可以不传
	 * @param sheetName 按sheetName加载  可以不传
	 * @param headRowNumber 从第几行开始读，因为会有表头的情况。可以不传，表示从第1行开始
	 * @return
	 *
	 */
	public static <T> void analysisExcel(String excelFilePath,ExcelListener<T> listener, Class<T> aa,Integer sheetNum,String sheetName,Integer headRowNumber){
		if(DataUtil.isEmpty(excelFilePath)) {
		}
		if(excelFilePath.toLowerCase().endsWith(".xls") && excelFilePath.toLowerCase().endsWith(".xlsx")) {
		}
		
		File excelFile=new File(excelFilePath);
		ExcelReaderBuilder builder=EasyExcel.read(excelFile, aa,listener);
		ExcelReaderSheetBuilder sheetBuilder=null;
		if(DataUtil.isNotEmpty(sheetNum)) {
			sheetBuilder =builder.sheet(sheetNum);
		}
		if(DataUtil.isNotEmpty(sheetName)) {
			sheetBuilder =builder.sheet(sheetName);
		}
		if(DataUtil.isEmpty(sheetName) && DataUtil.isEmpty(sheetNum)) {
			sheetBuilder =builder.sheet(0);
		}
		if(DataUtil.isEmpty(headRowNumber)) {
			headRowNumber=0;
		}
		sheetBuilder.headRowNumber(headRowNumber);
		
		sheetBuilder.doRead();
		
//		List<T> objList=listener.getDatas();
//		return objList;
	}
	
	
	
	
	
	
	public static void main(String[] args) {
		String excelFilePath="C:\\Users\\hchen\\Desktop\\康富总表.xlsx";
//		try {
//			List<Object> objList =analysisExcel(excelFilePath);
//			for (Object object : objList) {
//				System.out.println(object);
//			}
//		} catch (BusinessException e) {
//			e.printStackTrace();
//		}
		
//		ExcelListener<DemoData> listener=new ExcelListener<DemoData>();
//		EasyExcel.read(excelFilePath, DemoData.class, listener).sheet(0).doRead();
//		List<DemoData> objList=listener.getDatas();
//		for (DemoData demoData : objList) {
//			System.out.println(demoData);
//		}
//		
//		
//		//多sheet同时读取，提高性能，而不是每次重新加载一个sheet再读。
//		ExcelReader excelReader = null;
//        try {
//            excelReader = EasyExcel.read(excelFilePath).build();
//
//            // 这里为了简单 所以注册了 同样的head 和Listener 自己使用功能必须不同的Listener
//            ReadSheet readSheet1 =
//                EasyExcel.readSheet(0).head(DemoData.class).registerReadListener(new ExcelListener<DemoData>()).build();
//            ReadSheet readSheet2 =
//                EasyExcel.readSheet(1).head(DemoData.class).registerReadListener(new ExcelListener<DemoData>()).build();
//            // 这里注意 一定要把sheet1 sheet2 一起传进去，不然有个问题就是03版的excel 会读取多次，浪费性能
//            excelReader.read(readSheet1, readSheet2);
//        } finally {
//            if (excelReader != null) {
//                // 这里千万别忘记关闭，读的时候会创建临时文件，到时磁盘会崩的
//                excelReader.finish();
//            }
//        }
        
//        long start = System.currentTimeMillis();
//		ExcelListener<KfengineeringPlant> listener=new ExcelListener<KfengineeringPlant>();
////		ExcelListener<T> listener=new ExcelListener<T>();
//        try {
//			List<KfengineeringPlant> list = analysisExcel(excelFilePath, listener,KfengineeringPlant.class, null, null, 3);
//			long end = System.currentTimeMillis();
//			
//			System.out.println("解析耗时："+(end-start)/1000);
//			for (int i = 0; i < list.size(); i++) {
//				if(i<10) {
//					System.out.println(list.get(i));
//				}
//			}
//        } catch (BusinessException e) {
//			// TODO Auto-generated catch block
//			e.printStackTrace();
//		}
		
		
	}
	
	
	
}
